package com.zz.ai.component;

import org.springframework.ai.chat.model.ChatModel;
import org.springframework.ai.document.Document;
//import org.springframework.ai.transformer.KeywordMetadataEnricher;
import org.springframework.beans.factory.annotation.Qualifier;
import org.springframework.stereotype.Component;

import java.util.List;

//@Component
public class MyKeywordEnricher {


//    private final ChatModel chatModel;
//
//    MyKeywordEnricher(@Qualifier("ollamaChatModel") ChatModel chatModel) {
//        this.chatModel = chatModel;
//    }
//
//    public List<Document> enrichDocuments(List<Document> documents) {
//        KeywordMetadataEnricher enricher = new KeywordMetadataEnricher(this.chatModel, 5);
//        List<Document> result = enricher.apply(documents);
//        return result;
//    }

}
